import axios, { InternalAxiosRequestConfig, AxiosResponse } from 'axios'
import { useUserStoreHook } from '@/store/modules/user'
import { ResultEnum } from '@/enums/ResultEnum'
import { TOKEN_KEY } from '@/enums/CacheEnum'

// 创建 axios 实例
const service = axios.create({
    baseURL: import.meta.env.VITE_APP_BASE_API,
    timeout: 50000,
    headers: { 'Content-Type': 'application/json;charset=utf-8' }
})

// 请求拦截器
service.interceptors.request.use(
    (config: InternalAxiosRequestConfig) => {
        const accessToken = localStorage.getItem(TOKEN_KEY)
        if (accessToken) {
            config.headers.Authorization = accessToken
        }
        return config
    },
    (error: any) => {
        return Promise.reject(error)
    }
)

// 响应拦截器
service.interceptors.response.use(
    (response: AxiosResponse<any>) => {
        // 检查配置的响应类型是否为二进制类型（'blob' 或 'arraybuffer'）, 如果是，直接返回响应对象
        if (response.config.responseType === 'blob' || response.config.responseType === 'arraybuffer') {
            return response
        }

        const { code, data, message } = response.data
        if (code === ResultEnum.SUCCESS) {
            return data
        }

        if (code === ResultEnum.TOKEN_INVALID) {
            ElNotification({
                title: '提示',
                message: '您的会话已过期，请重新登录',
                type: 'info'
            })
            useUserStoreHook()
                .resetToken()
                .then(() => {
                    location.reload()
                })
        } else {
            ElMessage.error(message || '系统出错')
        }
        return Promise.reject(new Error(message || '系统出错'))
    },
    (error: any) => {
        // 异常处理
        if (error.response.data) {
            const { code, message } = error.response.data
            if (code === ResultEnum.TOKEN_INVALID) {
                ElNotification({
                    title: '提示',
                    message: '您的会话已过期，请重新登录',
                    type: 'info'
                })
                useUserStoreHook()
                    .resetToken()
                    .then(() => {
                        location.reload()
                    })
            } else {
                ElMessage.error(message || error.message || '系统出错')
            }
        }
        return Promise.reject(error.message)
    }
)

// 导出 axios 实例
export default service
